"""
Multi-Task Learning (MTL) Framework
基于自定义张量库的多任务学习框架
"""

from .config import MTLConfig
from .trainer import MTLTrainer
from .architectures import MTLArchitecture, HardParameterSharing
from .weighting import WeightingStrategy, EqualWeighting, GradNorm, UncertaintyWeighting
from .utils import MTLDataset, compute_task_metrics

__all__ = [
    'MTLConfig',
    'MTLTrainer',
    # Architectures
    'MTLArchitecture',
    'HardParameterSharing',
    # Weighting strategies
    'WeightingStrategy',
    'EqualWeighting',
    'GradNorm',
    'UncertaintyWeighting',
    # Utils
    'MTLDataset',
    'compute_task_metrics'
] 